package 数据结构与算法.稀疏数组;

public class SparseArray {
    public static void main(String[] args) {
        //创建一个原始的二维的数组
        //0：代表没有棋子，1表示黑子，2表示篮子
        int[][] chessArr1 = new int[10][10];
        chessArr1[1][2]=1;
        chessArr1[2][3]=2;
        //输出原始二维数组
        System.out.println("原始二维数组如下：");
        for (int[] row : chessArr1) {
            for (int data : row) {
                System.out.printf("%d\t",data);
            }
            System.out.println();
        }

        //将二维数组转为稀疏数组的思想
        //1、先遍历二维数组。得到非0数据的个数
        int sum=0;
        int rows=0;
        int cols=0;
        for (int[] value : chessArr1) {
            for (int i : value) {
                if (i != 0) {
                    sum++;
                }
            }
            cols = value.length;
            rows++;
        }
        //2、创建对应的稀疏数组
        int[][] sparseArr = new int[sum + 1][3];
        //给稀疏数组赋值
        sparseArr[0][0]= rows;
        sparseArr[0][1]= cols;
        sparseArr[0][2]= sum;

        //遍历二维数组，将非0的值存放到sparseArr中
        int count=1;//用于记录是第几个非0数据,即代表稀疏数组行的下标
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                if (chessArr1[i][j]!=0){
                    sparseArr[count][0]=i;
                    sparseArr[count][1]=j;
                    sparseArr[count][2]=chessArr1[i][j];
                    count++;
                }
            }
        }

        //输出稀疏数组
        System.out.println();
        System.out.println("得到的稀疏数组为：");
        for (int[] ints : sparseArr) {
            System.out.printf("%d\t%d\t%d\t\n", ints[0], ints[1], ints[2]);
        }
        System.out.println();

        //将稀疏数组恢复成原始二维数组
        //1、先读取稀疏数组的第一行，根据第一行的数据，创建原始的二维数组
        int[][] chessArr2 = new int[sparseArr[0][0]][sparseArr[0][1]];
        //2、再读取稀疏数组的后几行数据，并赋给原始的二维数组
        for (int i = 1; i < sparseArr.length; i++) {
            chessArr2[sparseArr[i][0]][sparseArr[i][1]]=sparseArr[i][2];
        }

        //输出恢复后的二维数组
        System.out.println();
        System.out.println("恢复后的二维数组如下：");
        for (int[] row : chessArr2) {
            for (int data : row) {
                System.out.printf("%d\t",data);
            }
            System.out.println();
        }
    }
}
